import torch.nn as nn
import torch


class HiDDeN(nn.Module):
    def __init__(self, encoder, decoder, noise_layers):
        super().__init__()
        self.encoder = encoder
        self.decoder = decoder
        self.noise_layers = noise_layers
